package a_doubleWrite;

import util.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * 129. 求根节点到叶节点数字之和
 * 给你一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。
 * 每条从根节点到叶节点的路径都代表一个数字：
 * <p>
 * 例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。
 * 计算从根节点到叶节点生成的 所有数字之和 。
 * <p>
 * 叶节点 是指没有子节点的节点。
 *
 * @author luolidong
 * @version 1.0
 * @create 2021-07-26 11:17
 */
public class L129 {
    List<Integer> list = new ArrayList<>();

    public int sumNumbers(TreeNode root) {
        if (root.left == null && root.right == null) return root.val;
        dfs(root, 0);
        int sum=0;
        for(int i:list){
            sum+=i;
        }
        return sum;
    }

    private void dfs(TreeNode root, int i) {
        if (root == null) return;
        i=i*10+root.val;
        if (root.left == null && root.right == null) {
            list.add(i);
            return;
        }
        dfs(root.left, i);
        dfs(root.right, i);
    }

    public static void main(String[] args) {
        L129 l129 = new L129();
        l129.sumNumbers(TreeNode.build());
    }
}
